import torch
import torchvision.models

# 方式一 保存模型
vgg16_false = torchvision.models.vgg16(pretrained=False)
torch.save(vgg16_false, "../vgg16/vgg16_false_method1.pth")

# 方式二 保存模型字典，官方推荐
torch.save(vgg16_false.state_dict(), "../vgg16/vgg16_false_method2.pth")